package com.vic.delayedqueue.config;

import org.springframework.amqp.core.*;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;

import java.util.HashMap;
import java.util.Map;

/**
 * rabbitmq的通用延时队列配置
 * 1 声明通用交换机、通用路由键、通用队列、通用队列绑定到通用交换机。
 * 2 在声明通用队列时，指定死信交换机、死信路由键。注意：这里不要指定队列的TTL。
 */
//@Configuration
public class DelayedQueue2Config {

    // 通用交换机
    public static final String COMMON_EXCHANGE = "common_exchange";

    // 通用队列
    public static final String COMMON_QUEUE = "common_queue";
    // 通用路由键
    public static final String COMMON_ROUTING_KEY = "common";

    // 死信交换机
    public static final String DEAD_EXCHANGE = "dead_exchange";
    // 死信队列
    public static final String DEAD_QUEUE = "dead_queue";
    // 死信路由键
    public static final String DEAD_ROUTING_KEY = "dk";


    // 声明通用交换机
    @Bean("common_exchange")
    public DirectExchange directExchange() {
        return new DirectExchange(COMMON_EXCHANGE, true, false);
    }

    // 声明通用队列，该队列不设置消息过期时间，有参数传入，灵活控制
    @Bean("common_queue")
    public Queue commonQueue() {
        Map<String, Object> arguments = new HashMap<>();
        // 设置队列的死信交换机
        arguments.put("x-dead-letter-exchange", DEAD_EXCHANGE);
        // 设置死信的 routing key
        arguments.put("x-dead-letter-routing-key", DEAD_ROUTING_KEY);
        // 设置消息超时时间，单位ms
//        arguments.put("x-message-ttl", 30 * 1000);
        return QueueBuilder.durable(COMMON_QUEUE).withArguments(arguments).build();
    }

    // 声明 通用队列 绑定 通用交换机
    @Bean
    public Binding commonQueueBindingCommonExchange(@Qualifier("common_queue") Queue queue, @Qualifier("common_exchange") DirectExchange exchange) {
        return BindingBuilder.bind(queue).to(exchange).with(COMMON_ROUTING_KEY);
    }

    // 声明死信交换机
    @Bean("dead_exchange")
    public DirectExchange deadExchange() {
        return new DirectExchange(DEAD_EXCHANGE, true, false);
    }

    // 声明死信队列
    @Bean("dead_queue")
    public Queue deadQueue() {
        return new Queue(DEAD_QUEUE, true, false, false);
    }

    // 声明死信队列绑定死信交换机
    @Bean
    public Binding deadQueueBindDeadExchange(@Qualifier("dead_queue") Queue queue, @Qualifier("dead_exchange") DirectExchange exchange) {
        return BindingBuilder.bind(queue).to(exchange).with(DEAD_ROUTING_KEY);
    }

}